#! -*- coding: utf-8 -*-
"""
@Info: config
"""

OPERATE_TYPE = [str(i) for i in [0, 1, 2, 3, 4, 5, 6, 9, 11,
                12, 13, 14, 15, 16, 17, 19, 20, 21, 22]]

# 服务根路径
bash_path = '/home/data/llm-ner'

# 推理配置
OPERATE_MODEL_CONFIG = {
    'pretrain_model_path': bash_path + '/model/chinese-roberta-wwm',
    'labels': OPERATE_TYPE,
    'class_num': len(OPERATE_TYPE),
    'model_path': '../model_saved/operate_type/20240821_211610/pytorch_model.bin'
}
